"""
检查ChromaDB集合信息，查询前5个文档
"""

from db_utils import VectorStore, Database

# 创建数据库连接和向量存储实例
db = Database()
vector_store = VectorStore(db)

print("\nChromaDB集合信息:")
print(f"集合名称: {vector_store.collection.name}")

# 查看ChromaDB中存储的文档元数据
print("\nChromaDB中存储的文档元数据示例:")
# 查询前5个文档
sample_docs = vector_store.collection.get(limit=5)
for i, (doc_id, metadata) in enumerate(
    zip(sample_docs["ids"], sample_docs["metadatas"])
):
    print(f"\n文档{i+1}:")
    print(f"ID: {doc_id}")
    print(f"元数据: {metadata}")
    print(f"ID类型: {type(doc_id)}")
    print(f"document_id字段: {metadata.get('document_id', '不存在')}")
